package com.doublethread.easymart.common.result;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 分页统一响应体
 *
 * @param <T> 数据类型
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class PageResult<T> {

    /**
     * 总记录数
     */
    private Long total;

    /**
     * 总页数
     */
    private Long pages;

    /**
     * 当前页码
     */
    private Long current;

    /**
     * 每页大小
     */
    private Long size;

    /**
     * 数据列表
     */
    private List<T> records;

    /**
     * 从MyBatis-Plus的Page对象创建分页响应
     */
    public static <T> PageResult<T> of(Page<T> page) {
        return new PageResult<>(
                page.getTotal(),
                page.getPages(),
                page.getCurrent(),
                page.getSize(),
                page.getRecords()
        );
    }

    /**
     * 创建分页响应
     */
    public static <T> PageResult<T> of(Long total, Long pages, Long current, Long size, List<T> records) {
        return new PageResult<>(total, pages, current, size, records);
    }

    /**
     * 从Page对象创建分页响应并进行类型转换
     */
    public static <S, T> PageResult<T> of(Page<S> page, Function<S, T> converter) {
        List<T> convertedRecords = page.getRecords().stream()
                .map(converter)
                .collect(Collectors.toList());

        return new PageResult<>(
                page.getTotal(),
                page.getPages(),
                page.getCurrent(),
                page.getSize(),
                convertedRecords
        );
    }

    /**
     * 创建空分页响应
     */
    public static <T> PageResult<T> empty(Long current, Long size) {
        return new PageResult<>(0L, 0L, current, size, List.of());
    }
}
